trainer/main.py